Pytorch中torch.full()，torch.ones()和torch.zeros()函数解析 |
您所在的位置：网站首页 › pytorch new_ones › Pytorch中torch.full()，torch.ones()和torch.zeros()函数解析 |
Pytorch中torch.full()，torch.ones()和torch.zeros()函数解析
|
一. torch.full()函数解析
1. 官网链接
torch.full()，如下图所示： 返回创建size大小的维度，里面元素全部填充为fill_value 3.代码举例输出形状为(2,3)的二维张量，里面元素全部填充为5 x = torch.full(size=(2,3),fill_value=5) x 输出结果如下： tensor([[5, 5, 5], [5, 5, 5]]) 二. torch.ones()函数解析 1. 官网链接torch.ones()，如下图所示： 返回创建size大小的维度，里面元素全部填充为1 3. 代码举例 a = torch.ones(2, 3) b = torch.ones(5) c = torch.ones(size=(3,4)) a,b,c 输出结果如下： (tensor([[1., 1., 1.], [1., 1., 1.]]), tensor([1., 1., 1., 1., 1.]), tensor([[1., 1., 1., 1.], [1., 1., 1., 1.], [1., 1., 1., 1.]])) 三. torch.zeros()函数解析 1.官网链接torch.zeros()，如下图所示： 返回创建size大小的维度，里面元素全部填充为0 3. 代码举例 a = torch.zeros(2, 3) b = torch.zeros(5) c = torch.zeros(size=(3,4)) a,b,c 输出结果如下： (tensor([[0., 0., 0.], [0., 0., 0.]]), tensor([0., 0., 0., 0., 0.]), tensor([[0., 0., 0., 0.], [0., 0., 0., 0.], [0., 0., 0., 0.]])) |
【本文地址】